# python3 detect.py --weights ./pspv5s.pt或其他模型 --source data/images --conf 0.25 --img-size 1024
python3 train_custom.py --data data/custom.yaml --cfg models/yolov5s_custom_seg.yaml \
 --batch-size 2 --epochs 100 --weights ./yolov5s.pt --workers 1 --label-smoothing 0.1 --img-size 832 --device 0